from langchain.prompts import (
    ChatPromptTemplate,
    HumanMessagePromptTemplate,
    SystemMessagePromptTemplate,
)

from langchain_deepseek import ChatDeepSeek

# 创建prompt
template = ChatPromptTemplate.from_messages(
    [
        SystemMessagePromptTemplate.from_template("你是{product}的客服助手。你的名字叫{name}"),
        HumanMessagePromptTemplate.from_template("{query}")
    ]
)

# 创建LLM
llm = ChatDeepSeek(model="deepseek-chat")

# 创建prompt
prompt = template.format_messages(
    product="云计价",
    name="达牛",
    query="你是谁"
)

print(prompt)
ret = llm.invoke(prompt)
print(ret.content)